import numpy as np

a = np.arange(5)
# 一维数组遍历出的都是元素
for i in a:
    print(i)

b = np.arange(20).reshape(4,-1)
print(b)

# 多维数组遍历，取出第一维度的所有元素，返回数组（原来维度-1）
for i in b:
    print(i)

# b.flat 把多维数组展开为一维数组
for item in b.flat:
    print(item)

b = np.arange(27).reshape(3,3,3)
print(b)

for item in b.flat:
    print(item)